/*
** Copyright (C) 2010 Calin Radoni
**
** This file is part of CHScanner.
**
** CHScanner is free software: you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation, either version 3 of the License, or
** (at your option) any later version.
**
** CHScanner is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with CHScanner. If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once
#include "ScanMethod.h"
#include "ScanMethodArp.h"
#include "ScanMethodSyn.h"
#include "ScanMethodDns.h"

#include "ScanMethodHuntParams.h"

class ScanMethodHunt : public ScanMethod
{
private:
	ScanStartParameters sspc;
	ScanMethodArp scanMethodArp;
	ScanMethodSyn scanMethodSyn;
	ScanMethodDns scanMethodDns;

protected:
	bool dataProcessed;

	virtual int InternalInit(void);
	virtual int InternalExecute(void);

	DWORD myIP;
	virtual void ProcessArpAny(BYTE*, int);
	virtual void ProcessTcp(BYTE*, int, BYTE*);
	virtual void ProcessIcmpAny(BYTE*, int, BYTE*);

	MFullAddress addrX;
	MMachineInfo machX;
	MPortInfo    portX;

	void AddMachineAndPort(BYTE*, BYTE*, WORD, WorkMethodType);

	void Hunt(MMachineInfo*);

public:
	ScanMethodHunt(void);
	~ScanMethodHunt(void);

	ScanMethodHuntParams xHuntParams;
};
